# encoding: utf-8
# !/usr/local/bin/python3

"""
@author: Gao Shuo
@contact: dorothy400@163.com
@software: PyCharm
@file: eval.py
@time: 2020/3/7 12:59

input: 
output: 
"""


import Model
import dataset_process
import tensorflow as tf
from tensorflow.python.platform import app
from sklearn.model_selection import train_test_split
import time
import os


# x训练模型和保存
num_class = 100
batch_size = 32
capacity = 256 # 内存存储文件数

FLAGS = app.flags.FLAGS
app.flags.DEFINE_string("data_dir", './data/train', "Data storage files")


X, y = dataset_process.get_file(FLAGS.data_dir)  # 获取图片列表和标签列表
X_train, X_valid, y_train, y_valid = train_test_split(X, y, test_size=0.25, random_state=0)
image_batch, label_batch = dataset_process.get_batch(X_valid, y_valid, 224, 224, batch_size, capacity)













